from myenv import MyEnv
import time
import numpy as np

if __name__ == '__main__':
    myEnv = MyEnv(
        screen_size=(200, 200)
    )
    myEnv.reset()

    done = False
    
    while not done:
        actions = myEnv.sample_action()
        obs, reward, done = myEnv.step(actions)
        print(obs, reward, done)
        time.sleep(1)